/*
 * Copyright 2016-2019 yoara
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package yhao.infra.web;

import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.ComponentScan;
import org.springframework.context.annotation.Import;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
import yhao.infra.common.CommonBaseConfig;
import yhao.infra.web.filter.CrossOriginFilter;
import yhao.infra.web.filter.EncodeFilter;
import yhao.infra.web.filter.InnerAccessCheckFilter;

/**
 * Created by yoara on 2017/6/8.
 */
@ComponentScan("yhao.infra.web")
@Import({CommonBaseConfig.class})
@EnableConfigurationProperties(YhaoInfraWebProperties.class)
public class WebBaseConfig implements WebMvcConfigurer {
    @Bean
    public FilterRegistrationBean crossOriginFilterBean(YhaoInfraWebProperties properties) {
        boolean openFilter = !properties.getEnv().isForbidCrossOrigin();
        FilterRegistrationBean registration = new FilterRegistrationBean(new CrossOriginFilter(openFilter));
        registration.addUrlPatterns("/*");
        registration.setOrder(2);
        return registration;
    }

    @Bean
    public FilterRegistrationBean innerAccessCheckFilter(YhaoInfraWebProperties properties) {
        boolean check = properties.getEnv().isInnerAccessCheck();
        String ipPrefixs = properties.getEnv().getInnerAccessIpPrefix();
        FilterRegistrationBean registration = new FilterRegistrationBean(new InnerAccessCheckFilter(check,ipPrefixs));
        registration.addUrlPatterns("/*");
        registration.setOrder(2);
        return registration;
    }

    @Bean
    public FilterRegistrationBean encodeFilterBean() {
        FilterRegistrationBean registration = new FilterRegistrationBean(new EncodeFilter());
        registration.addUrlPatterns("/*");
        registration.setOrder(2);
        return registration;
    }
}
